from ultralytics import YOLO

# 加载预训练模型
model = YOLO('yolov8x.pt')

# 训练模型
results = model.train(
    data='coco8.yaml',  # 数据集配置文件路径
    epochs=100,  # 训练轮数
    imgsz=640,  # 输入图像尺寸
    batch=16,  # 批次大小
    name='yolov8_end'  # 训练结果保存的名称
)